-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Add more support for mxnet_to_coreml #14222
Conversation
* Add PReLU * Add group support in Convolution * Skip _minus_scalar and _mul_scalar
@pracheer could you help take a look? Thanks. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the change. It'll be great to have some unit tests for the cases you are adding.
Also added a couple of extremely minor (nitpick) comments.
* Add spare line between comment and code * Add two spare lines between functions * Change "Convert an leakyrelu" to "Convert a leakyrelu"
* Add test_tiny_prelu_leakyrelu_random_input * Add test_really_tiny_conv_random_input_multi_group * Add test_tiny_conv_random_input_multi_group * Modify test_conv_random by adding num_group
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for adding the test cases.
Couple of relatively minor comments otherwise the change looks good.
* Add support of "leaky" and "elu" * Add unit test for "leaky" and "elu"
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the changes!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* Add support of mxnet_to_coreml * Add PReLU * Add group support in Convolution * Skip _minus_scalar and _mul_scalar * Comply with code style * Add spare line between comment and code * Add two spare lines between functions * Change "Convert an leakyrelu" to "Convert a leakyrelu" * Check if num_group exists in param before assigning value * Add test cases for test_mxnet_converter * Add test_tiny_prelu_leakyrelu_random_input * Add test_really_tiny_conv_random_input_multi_group * Add test_tiny_conv_random_input_multi_group * Modify test_conv_random by adding num_group * Rollback to not setting num_group * Add "leaky" and "elu" in converter * Add support of "leaky" and "elu" * Add unit test for "leaky" and "elu"
* Add support of mxnet_to_coreml * Add PReLU * Add group support in Convolution * Skip _minus_scalar and _mul_scalar * Comply with code style * Add spare line between comment and code * Add two spare lines between functions * Change "Convert an leakyrelu" to "Convert a leakyrelu" * Check if num_group exists in param before assigning value * Add test cases for test_mxnet_converter * Add test_tiny_prelu_leakyrelu_random_input * Add test_really_tiny_conv_random_input_multi_group * Add test_tiny_conv_random_input_multi_group * Modify test_conv_random by adding num_group * Rollback to not setting num_group * Add "leaky" and "elu" in converter * Add support of "leaky" and "elu" * Add unit test for "leaky" and "elu"
* Add support of mxnet_to_coreml * Add PReLU * Add group support in Convolution * Skip _minus_scalar and _mul_scalar * Comply with code style * Add spare line between comment and code * Add two spare lines between functions * Change "Convert an leakyrelu" to "Convert a leakyrelu" * Check if num_group exists in param before assigning value * Add test cases for test_mxnet_converter * Add test_tiny_prelu_leakyrelu_random_input * Add test_really_tiny_conv_random_input_multi_group * Add test_tiny_conv_random_input_multi_group * Modify test_conv_random by adding num_group * Rollback to not setting num_group * Add "leaky" and "elu" in converter * Add support of "leaky" and "elu" * Add unit test for "leaky" and "elu"
Description
Add more support for mxnet_to_coreml
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Comments